hw2 cleaned up comments