assignment2 - problem1