Added code and updates to README