fix memory allocate issue
fix memory allocate
Merge pull request #1 from ruoxi-wang/master

Updated version