Fixed something I got wrong in hw1