Fixing up plot