Made the plot look nicer