plot with legend python