Create visualization.py

Visualize the initial data.