First commit of Problem 4