Bugs fixes & AI fixes