Fixing final plot