Removed commented code for __main__ in plot.py