Added l2 parameter to grid search, warm restarts, better output