add the code on readme