added comment functionality, seems to work, better color palatte for graphs