diff --git a/build.py b/build.py
index ef1be9e..ca9c552 100644
--- a/build.py
+++ b/build.py
@@ -6,9 +6,10 @@ import PyInstaller.__main__
 
 # Initialize constants
 SCRIPT = "cod_api_tool.py"
-ICON = "assets/icon.ico"
 NAME = "cod_api_tool"
-DIST_PATH = "bin/build"
+DIST_PATH = os.path.join("bin", "build")
+ICON_WIN = "assets/icon.ico"
+ICON_LINUX = "assets/icon.png"  # Linux uses .png for icons
 
 # Get absolute paths to data files
 script_dir = os.path.abspath(os.path.dirname(__file__))
@@ -20,30 +21,41 @@ if not os.path.exists(replacements_json):
     print(f"ERROR: {replacements_json} not found. Make sure this file exists.")
     sys.exit(1)
 
+# Determine system platform
+is_windows = sys.platform == 'win32'
+
 # Activate the virtual environment
-if sys.platform == 'win32':
+if is_windows:
     venv_activation_script = os.path.join(os.getcwd(), 'venv', 'Scripts', 'activate')
 else:
     venv_activation_script = os.path.join(os.getcwd(), 'venv', 'bin', 'activate')
-subprocess.call(venv_activation_script, shell=True)
 
-# Run PyInstaller
-PyInstaller.__main__.run([
+# Use the correct icon based on platform
+ICON = ICON_WIN if is_windows else ICON_LINUX
+
+# Construct PyInstaller command
+pyinstaller_args = [
     SCRIPT,
     '--name', NAME,
     '--noconfirm',
     '--onefile',
     '--console',
-    '--icon', ICON,
     '--distpath', DIST_PATH,
-    # This is the correct way to add the data file - preserve the directory structure
-    '--add-data', f"{charset_normalizer_data};charset_normalizer/assets",
-    '--add-data', f"{replacements_json};data"  # Note: using 'data' as the destination folder
-])
+    '--add-data', f"{charset_normalizer_data}{os.pathsep}charset_normalizer/assets",
+    '--add-data', f"{replacements_json}{os.pathsep}data",
+]
 
-# Clean up the build directory and spec file
+# Add icon only if it exists
+if os.path.exists(ICON):
+    pyinstaller_args.extend(['--icon', ICON])
+
+# Run PyInstaller
+PyInstaller.__main__.run(pyinstaller_args)
+
+# Clean up build directory and spec file
 shutil.rmtree('build')
 os.remove(f'{NAME}.spec')
 
-print(f"Build completed successfully. Executable is in {DIST_PATH}/{NAME}.exe")
-input("Press Enter to continue...")
\ No newline at end of file
+# Notify user where the output is
+executable_name = f"{NAME}.exe" if is_windows else NAME
+print(f"Build completed successfully. Executable is in {DIST_PATH}/{executable_name}")