Changed a bit of the arm-classify script