Python program to write first n prime number in text file

prime=[]
limit=int(input("How many prime numnbers you want to write : "))
n=2
count=0
while count<=limit:
    flag=True
    for a in range(2,n):
        if n%a==0:
            flag=False
    if flag==True:
        count=count+1
        prime.append(str(n)+"\n")
    n=n+1
fp=open("prime.txt","w")
fp.writelines(prime)
fp.close()