Added work controller and fixed some coloring.