Updated the initialisation step