initial draft code for classification