assignment1 - problem2