added image to data transformer model class