Changed the code/put in the equations.

Now I just need to see if it will work...