Model Class Little Refactoring

Signed-off-by: vukis <vukis@bk.ru>