Modify initalize logic.