Include the CLI class so we can use it

At the moment using the rake task according to the example fails, due to the `CLI` class not being required.

This fixes that.