Added more tests for package