Added coloring
Added support for more keywords
Changed output to a colored graph