Small bug fixes
Updated readme