revised the math problem