updated a function

let me commit please