updated prediction code to use sklearn (much cleaner than my homegrown code)