diff --git a/weather-server-python/weather.py b/weather-server-python/weather.py
index 74a3015..5822b4e 100644
--- a/weather-server-python/weather.py
+++ b/weather-server-python/weather.py
@@ -89,6 +89,9 @@ Forecast: {period['detailedForecast']}
 
     return "\n---\n".join(forecasts)
 
-if __name__ == "__main__":
+def main():
     # Initialize and run the server
-    mcp.run(transport='stdio')
+    mcp.run(transport='stdio')    
+
+if __name__ == "__main__":
+    main()