Added way to add a class per model