Adding additional stuff for Python code