0.3.1

Small fixes