Collab for Project 2: https://colab.research.google.com/drive/1Ar36W2W0CNRHbRpx6VZAwIjAe_WQ5jN8?usp=sharing