Added vis.py script to create visualization of results