I think I have fixed the math