Add assignment 1 code and class diagram