Finished Assignment 2, edited comment on line 59