Added python solution for problem #3