Separate into classes. More pylint.