Create class_vis.py