Updated init script to take command param