added first draft of models