Add code to README