Add support for pretty-printing of return values as well