Finish task controller spec and update task controller