Minor patch to rectify behavior in 0.4.0
Bumping version to 0.4.1