Improvement | add rspec tests