Add support for running in docker