Tested on local branch, tests passing. Removed puts.