Removed puts at end to test with rspec