Added level restriction support