Add close function