Added a few tests for more coverage.