Adding in custom test for CLI extra credit