Change command to entrypoint, update run.sh to take args