assignment-1: problem 1 completed