Add prediction based context scheduling mechanism