Switched rspec syntax to expect style
Added ability to run callback on stub invocation