added data_viz.py and updated utils.py