missing import
fix plotting's use of legends