Update README and code accordingly
Update README and code accordingly