updated the task to accept arguments