This is the code for the animation that I had running in this video: https://www.youtube.com/watch?v=JoJUKjjmyQg&t=59s
To run this code you need python3 or greater and the p5 python module.
p5 installtion guide: https://p5.readthedocs.io/en/latest/install.html