trying supervised learning