Version bump to 0.0.4
Fixing minor issues