Tidied up logic in the Learn Attack Controller