some minor code updates