Move functionality to the install script