improving the code format