Fix for compare_models.py

Everything needs to be contained within main if the script is to be called as compare_models instead of python compare_models.py