tidy up the code format