Broke the code
Revert "Broke the code"

This reverts commit 2cc4290c2a4d6afc9df483f149a6763a646e183d.
Added a pickle file for visualization