update README and improve final print messages