Version 0.3.1

- Fix install problems with imports