Update init script: making it more generic