More work on the level editor