Update init function