Fixing problem 4