[update:] adding tests