Fixed bug on update (again)
0.0.4
Added unit tests