How to distinguish between plots in python