Cleaned-up the Load_model module.