An improved AI implementation for SOF2