add stuff for assignment2