switch to use docker module instead of custom code