Adding support for init script