Adding in unit tests for comman