Add in legend for GO plot