Add ability to start and stop a task