Changed color plotting to list comprehension