Yes, this can be done through in the staff’s profile under the details tab. 

The colour coding will show on timesheets and schedules.