Add two Python files.
Improve code.