Added argument to scheduler