Updated predictive_text.py to return an array rather than printing the words...