changed color function to match readme