Updated README
Improved code robustness.