Extract whole thing to class
Add loading screen when switching scenario