Assignement 1 - Problem 4 - Fix some mistakes