Add ploting code and plots