Update Gemfile
Add tests for generate cmd
Clean up generate cmd