Convert raw pointers to std::shared_ptr or std::unique_ptr

[trungtle]

We're using raw pointers everywhere. Could potentially bite us with memory leak