Update Gem

Add test coverage
Remove puts on test