Updated snippet for def main

nvim/UltiSnips/python.snippets

@@ -7,12 +7,12 @@ snippet date
 endsnippet
 
 snippet ifmain
-def main():
-    """$1"""
+def $1():
+    """$2"""
     return
 
 if __name__ == '__main__':
-    main()
+    $1()
 endsnippet
 
 snippet logger