Adding small changes to LevelFactory