More steps in main.py
minor progress on test_model.py