Fixed a few issues with dirty_map.py and parallel_dirty_map.py