Refactore Class