Add axis function example
minor fixes