v0.2 - Bug fixes