fixing bugs v2