tidy up codes
fix typos