Minor improvments to README
Aligned source code with README :)