Add partial function and test