Edited Gem

It now return the actual output from the code.