problem 4 - edit