Bug fixes - v0.3.1