Refactor custom class into another file.