Add classify function + Add model evaluation