Changing some code formatting for week-5-assignment-4