minor code updates