Added features to the level editor