fix aesthetic look of the code