made changes to foo.py (plotting) but still WIP