Python Close File After Writing