Challenge problem from problem set 1 for Intro to NLP