Update Dockerfile

Fix #2