Trying fix problem 2