Add initialisation code etc