cleanup and add print color method