Solved 2 and 3 in python