Added axes specification to tests and example