Adding whitespaces.
Adding comments.
Adding new-lines using `\n` rather than `print()`
Change to `os` module.