Minor refactor
Release 0.1.1