Added prototype in python for problem 3