Enhance rspec functionality