Add typing information for axes