Added print.
Removed free bug.

Freeing device mem instead of host mem.