replaced axis geometry with code