Add all the necessary code
Update README