Fix comments
Merge branch 'feature/work_on_models' into develop