Updating classifier unit tests