Small updates and fixes
0.1.0