Problem 4 solution and small update to readme.md