removed import
added some sklearn stuff
changed formatting
fixed an issue with an undeclared variable