Stage 1: Classification