added color feature
polish