Refactor: scheduler: functionize initializing working set

... for code isolation and readability. A slight change in behavior is that the
working set reset occurs at the beginning of a new scheduler run rather than
the end of the previous one.