fix on classifying images