Add spec for generator
Update guard to run spec correctly