Remove dependency of HeapScheduler in Model class