Python Plt Color Code