Changes to level system to get it working