- adding some code for loading and visualization.