Refactore class