Recreated model to work with scikit-learn. Great stuff