Pune Test: Australia take firm control