Use `with` Python idiom

[mallamanis]

Hi, minor suggestion that might reduce memory consumption. Try using the with open(x, mode) as f: context manager. This might help with reducing the memory consumption since it closes the files when you don't need them (it's also a good Python coding practice). See here for more.

Noticed this here and the rest of the code below.