Fix #2 and bump to 0.0.4