AI 'Working', added TODO