Problem 4 equations check and edit