assignment2 - problem 1 & 2