Update baseline.py

Changed the way we build matched_words list to improve the score