Add RSpec to write tests