Adding more unit tests for commands