Update to fix annotations of the axes