properly comparing all of the models to my models for the 338
saving images to img dir now