fix argument naming
Update the docs and update Dockerfile(todo)