def find_king(game_board, bw):
    """
    Finds right king
    :param game_board: array of board
    :param bw: black or white to move?
    :return: Square with king, None if king isn't present
    """
    # Find king
    for r in game_board:
        for sqr in r:
            if sqr.get_content() is not None:
                if sqr.get_content().get_type() == "K" and bw == "w":
                    return sqr
                elif sqr.get_content().get_type() == "k" and bw == "b":
                    return sqr
    # Didn't find king. Will hopefully never reach this
    return None