Improved UI and level editor