Changes to make pretier the plot