Adding more test + coverage
adding more tests for command