style: tidy up codes