Cleaned up the code and fixed the colors a bit. Feel free to tweak.