now starting writing the classification functionality