re-indent code a little bit