update to 0.0.4
refactor code