Added support for non-cartesian bases and visualisation of axes