Added first draft of model. TODO: test