Add initialize code