Bump to v0.4.1
Update tests