Correct solution for problem 005