Tweaks to firnmodel.py