minor update to functions