updated model_validation.py to include plotting