Updated example and primitive shader

- The example now shows the X, Y, and Z axis.
- Fixed a minor bug in the primitive shader.