added the code, few tweaks to the readme