Minor updates (v0.1.1)