Add minor code