run python main.py for a nice plot