How do I wait for the function to complete