add generator and fix specs
add one more spec