Add some tests for the command line tool