foo.py updated to deal with masks