Fixed grid_search and RDD issues