cleaned up the code

[sdushantha]

Hey,

Really cool script!

I made some minor changes to your script to make it a little cleaner :blush:

• removed trailing white spaces
• organized imports
• removed unused import (time)
• use with open() as f:

[sameera-madushan]

Thanks for the contribution ♥