Improve inline comments for better type hint.
Small bug fix.
Increment version to 0.5.1