Fixed solution.

Changed output, found memory leak.