Python - Changed to draw more graphs.