Added very simple solution for problem 4