Add title comment to plot.py