Solution to ps1

[jetamors]

The solution I wrote is behind the cut. Please let me know if you have any questions about the math or the code itself, or feel free to share your own solution in the comments.

( ps1.py )