changed plotting to look cooler