Updated to use expect() rspec syntax