touched up the controller
small change to task controller