Specific AI added. Still a bit buggy