Add classify button back in.