Added class in model