Add classify method