Update README

Adding new code