fixed solution 1
initial solution for problem 3