Small refactor of run task
Merge pull request #2 from vsalaman/task_run

Small refactor of run task