wrap it inside function