Adds python solution to problem 4.