Small refactor
Bump to 0.0.4