Start adding scenario generation code