def add_sentiment_to_df(df, doc_ids, sentiment_dict):
    """
    Input - 
            df - Dataframe passed by get_[tweet/headlines]_sentiment
            doc_ids - UUID for docs
            sentiment_dict - Dict mapping doc_ids to sentiment dict
    Returns - DF with sentiment columns
    """

    # Initialize lists for each polarity
    neg, pos, neu, compound = [], [], [], []

    # Populated above lists
    for i in range(len(df)):
        row = df.iloc[i]
        sentiment = sentiment_dict[doc_ids[i]]
        neg.append(sentiment['neg'])
        pos.append(sentiment['pos'])
        neu.append(sentiment['neu'])
        compound.append(sentiment['compound'])
    
    # Add to DF
    df['neg'] = neg
    df['neu'] = neu
    df['pos'] = pos
    df['compound'] = compound

    return df