Fix formatting issues for Assignment 3