Add code, update readme