add main to python snippets