Create assignment_plotting.py