Modifications to the python analysis package distributed with MITgcm