Bugfix and CMakeLists added

[ciaxj88]

Hello. I got your code, and I fixed the part that I think were not correct.

I would be very happy if you can update the code with that.

Correction: Removed memoryleak (all the malloc) Fixed the warnings Add the headers for correct import

Thanks for creating this repository, it really saved my life