Adds small improvements in the code and README.