Discovered that Rodrigo sucks at math. Fixed his math.

Updated test.py to contain better math