Machine learning practice in python using sklearn