Stying up a bunch of things
Added the legend
Cleaning up everything to try to make it faster
Cleaned up everything