def remove_after_stop(sentence):
    """Remove all tokens after the stop token"""
    try:
        end_index = sentence.index('.')
    except ValueError:
        end_index = None
    if end_index is not None:
        if len(sentence) > end_index + 1:
            sentence = sentence[:end_index + 1]

    return sentence