Add a helper function