Fix issues with cylon generate module.

This fixes #11