added tests
minor change