move models to separate module