Minor updates to problem 2 formatting