Improving code on README