Fixed import error and added Dockerfile

Devin and I worked on the issue that this repo didn't work due to some import error. ( #2 )

This PR addresses the import issue and supports Docker containerization.