Fix package import and bump to 0.3.1