Update to 0.1.1 - Small improvements & comments