Updates to reflect changes in v0.1.1