Improving comments and readme