Small change to begin adding legends