Cleaned up some level select code

Thought the for-loop didn't work before but it does now.