Refactor Level and LevelScreen