Fixed Input Code

As a result made the level selection nicer