Removed unnecesary imports, file opened as 'r'
Fixed minor bug
Updated readme