Change arguments.
Wrap up classes to one. And refactor code.