Added level selection back in