Add rspec
Add generate function