Problem 1 assigment 2