Imp documentation
Fix tests
Fix grid_search.py example