Update README.md
Cleaned up code and tweaked colour mapping further