small edit

small edit to functions