Fixed imports. Thanks to Bas van Dijk for the heads-up.
Bumped version to 0.4.1