Improve code on readme file