Australia snapped up two wickets in the final session on Day 2 to leave India with work to do in the third Test as they chase the hosts' 338 following Steve Smith's first century in more than a year.