Added license headers to .py files

Added license headers to the two Python files.