Remove color, need to be rewritten