Add support for axis orientation