Assignment 1 DONE, everything works