Improved one unit test
Updated grid search to account for the sklearn scoring problem I found yesterday