Refactor screen transitions.
Little steps to level editor