small change to functions