Added image classifier, modified base creator