Minor clean-up of investigation.py