Change to get nice plots