[Minor] Improve tests. Closes #24.
- also fixing couple of typos