Refactor code
Release 0.4.0