Cleaning up the comments and adding tests to commands.go