Updates script to do classification (hopefully)