Why import `os`? It doesn't look like it's being used anywhere in this file.