minor code tweeks