Small typo in prepare_for_plotting.py fixed.