Fixed issues with cli script
Bump to v0.6.1