Added some level logic