Added necessary kwargs to task method