Update to version 0.1.0
A little refactor to cleanup style