diff --git a/connect-to-wifi/main.go b/connect-to-wifi/main.go
index 1ab4807..5675214 100644
--- a/connect-to-wifi/main.go
+++ b/connect-to-wifi/main.go
@@ -24,6 +24,7 @@ func dhclient() {
 	for {
 		fmt.Print(" should check progress here")
 		time.Sleep(20000 * time.Millisecond)
+		return
 	}
 }
 
diff --git a/get-image/main.go b/get-image/main.go
index 327e5ee..5d7d905 100644
--- a/get-image/main.go
+++ b/get-image/main.go
@@ -82,6 +82,7 @@ func succeeded () {
 	log.Println("")
 
 	shell.Run("dialog --ascii-lines --msgbox 'Flash_Image_Completed' 10 70")
+	shell.Run("poweroff")
 
 	os.Exit(0)
 }