updated the assignment1 class