bug fixes and v2