Minor aesthetic updates

- Removed unused import
- Moved smart_open to the bottom
- Added and rearanged comments