we need class

It works but we need class