Solution 4
Updated readme