Tweaks tests a bit.
Update changelog
Bump version to 0.3.1