Fixed the math problem