Update to 0.6.1

Fix the release