Add scheduler functionality