A few tweaks and readme