Added level class
Made changes to handle level class