Minor fixes
Updated version to v0.5.1