Modified the classification code and also the run.sh script