fixed bug in problem1 again