Solved contest problems in python!