Cleaned up some warning, and updated the code in standard_nonlin_model.py