Added classify method