Add unit test for version command