oops, I lost some changes on image_classification. style improvements