Add a few tests and fix reverse_words