added a class to the code