Add meaningful tests
Update Tests

- More tests
- Option to return only some words from get_stats
- Fix no words bug
- More tests