solved till problem 2.1