Usman Khawaja tucks the ball away © Associated Press

Tom Latham muscles the ball over the leg side © Getty Images

A short ball from Kasun Rajitha had Ajaz Patel in trouble © Getty Images

Usman Khawaja brings up his half-century © Cricket Australia/Getty Images

Henry Nicholls and Tom Latham shared a 114-run partnership for the fourth wicket © AFP

Tom Latham walks off the field, having carried his bat through for a remarkable 264 not out © Associated Press

Tim Paine takes off for a run © Cricket Australia/Getty Images

Kasun Rajitha appeals vociferously for a wicket © Associated Press

Things heated up between Virat Kohli and Tim Paine in the first session © Getty Images

Tom Latham broke records on the third day © ESPNcricinfo Ltd

Virat Kohli and Tim Paine exchanged a few words © Getty Images

Jasprit Bumrah goes up in appeal © Getty Images

Usman Khawaja and Tim Paine extended Australia's lead © Cricket Australia/Getty Images

Colin de Grandhomme waltzed away to a 53-ball 49 © Getty Images

Tom Latham racked up his maiden Test match double century © Getty Images

Lahiru Kumara is chuffed after taking out BJ Watling for a duck © Getty Images

Tim Paine ducks under a bouncer © Getty Images

Chairman of selectors Trevor Hohns and coach Justin Langer have a chat © Getty Images