code clean up

got code coloring to work