Nicer function snippet for python

UltiSnips/python.snippets

@@ -19,14 +19,15 @@ def ${1:function}(${2}):
 params = t[2].replace(", ", ",").split(",")
 snip.rv = ""
 for param in params:
-	snip.rv += f"@param {param}"
+	snip.rv += f"\t\t@param {param}\n"
 `
 	Returns:
-		$4 ${5:description}
+		$4
 	"""
-	$6
+	$5
 	`!p
 if t[4] != "":
 	snip.rv = f"\n    return {t[4]}"`
+
 $0
 endsnippet