Add test command for testing the new function