feat: Dockerfile and more flexible optimize input