many updates to classify script and plotting