Adding basic plotting code