Minor edits to plot_score_by_weight.py