Adding classify code