Problem 6 solved but not sure if ok