Edit model6.py