Add code coloring to readme

Totally understand if this kind of this isn't wanted. Personally, I think it makes reading the code in the readme much easier.