added some nicer plot