adding test as well