!5 edit readme

Merge pull request !5 from kou he/master

main.py

@@ -20,6 +20,8 @@ while True:
             else:
                 print("Wrong password!")
                 wrong_time += 1
+                # print you had wrong password wrong_time times, max 3 times
+                print("You had wrong password %d times, max 3 times" % wrong_time)
                 if wrong_time == 3:
                     print("Too many wrong password!")
                     exit()
@@ -28,7 +30,7 @@ while True:
     elif input_number == "2":
         input_username = input("Enter a username\n")
         input_password = input("Enter a password\n")
-        # check if the username is in the dictionary so it is duplicate
+        # check if the username is in the dictionary  it is duplicate
         if input_username in username_and_password:
             print("The username is duplicate!")
         else:
@@ -65,12 +67,11 @@ while True:
                 # add the username and password to the dictionary
                 username_and_password[input_username] = input_password
                 print("Register successfully!")
-                print("Auto login...")
-                exit()
+                # back to menu
+                continue
 
     else:
         print("Wrong number!")  # if the input is not 1 or 2, it will print wrong number
     os.system("pause")
     os.system("cls")
     # clear the screen
-