Fix Dockerfile
Add tests