Added plots and refactored cross_main.py