Resolved issue with registering a function to also make certain we add the function