Add code to add vertical lines in the plot