Moved difficulty selection code to create_engine