Updated level class