fixing problem 2