Add sub function to remove duplicated code