Improve benchmark code
Bump version to v0.1.1