update comments and functions in tests