Tested my code with RSpec