Added messy code for legend

Added legend and tweaked some colors