modify some logic
add number to rspec test case