v0.1.1

Fixing new architechture and bugs that comes with it