Added Python solution for problem #4