Corrects mount.py to match the assignement