Fixed some failing tests
Version bump to 0.3.1