Put code in class