Improvements to get-baseline.py.