Latest plotting code added