add some class in model