Refactor TaskScheduler

Signed-off-by: Oleg Bulatov <oleg@bulatov.me>