Adds class. But need to fix it so that it gets it from database.