Add initialize function