problem number 2 fixed