Had to implement a

    if not warrior
      return nil
    end

in the play_turn method, to workaround a bug. But other than that, this is great!