add color, clean up code
add a readme