diff --git a/Dockerfile b/Dockerfile
index 5615f96..eae1fd4 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-FROM golang:1.16.2
+FROM golang:1.16.2 AS builder
 
 RUN mkdir -p /trisongulate
 WORKDIR /trisongulate
@@ -7,4 +7,9 @@ ADD . .
 RUN go mod download
 RUN CGO_ENABLED=0 GOOS=linux GOARCH=amd64 go build
 
+FROM ubuntu:latest
+
 ENTRYPOINT [ "/trisongulate/trisongulate" ]
+WORKDIR /root/
+COPY --from=builder /trisongulate/trisongulate .
+CMD ["./trisongulate"]  
\ No newline at end of file