Add level selection feature