Made level selection work