Edited program to test all axes

Signed-off-by: Tushar Pankaj <tushar.s.pankaj@gmail.com>