Allow users to pass in a single task, add test for single task