Add test file and complete the cmd py