Update install.py

install.py

@@ -5,6 +5,10 @@ linux_packages = [
     'zsh'
 ]
 
+linux_after_command = {
+    'zsh': 'chsh -s $(which zsh)'
+}
+
 os_type = platform.uname().system.lower()
 
 is_linux = os_type == "linux"
@@ -31,6 +35,8 @@ def install(package):
     else:
         print(f"Install not supported for: {os_type}")
         exit(1)
+    if is_linux and package in after_command:
+        os.system(linux_after_command[package])
 
 def check_os():
     if is_linux or is_macos or is_windows: