Put it in a class