First draft except model selection
Final first draft