Fixing plotting hopefully.