Add code and update Readme