change to a function i think