polish code and clean README