v0.1.1

Fixing issues