Cleaning up the level code