add class from the previous version