* clean up model classes

Signed-off-by: Dmitry Ponyatov <dponyatov@gmail.com>