Add support for docker