Added colors in print state, shorten code.