I think you need to pass a function in there