Fixed formatting for homework 2