Added Python solution for problem 4