Small bug fix
Updated README