Use Python to solve math questions (1)