Python Close File After Write