Updates to match docker container