Better formatting for the 3rd problem

Round two :)