Forgat to add class