cleanup code

src/example2/function.py

@@ -131,15 +131,15 @@ def manage_memory(memory_size):
     except ValueError:
         return jsonify({'error': 'Invalid memory size'}), 400
 
-    if size:
+    if size > 0:
         if stress_process:
             # Kill existing stress-ng process
             subprocess.run(['pkill', 'stress-ng'])
             stress_process = None
 
         # Start stress-ng process with specified size
-        stress_process = subprocess.Popen(['stress-ng', '--vm', '1', '--vm-bytes', size])
-        return f'Started new stress-ng process with {size} memory size.'
+        stress_process = subprocess.Popen(['stress-ng', '--vm', '1', '--vm-bytes', memory_size])
+        return f'Started new stress-ng process with {size}MB memory size.'
     else:
         if stress_process:
             # Kill existing stress-ng process