how to create legend in python