Updated to use the docker hub image directly