Created a better interface with sklearn library
Created better interface with sklearn library