working on refinig the plot