Add parameter to add_partial to avoid having two functions