Adds other unit tests (should be good). Closes #6