re-organizing data and example files. Added plotting script in visualize.py