format code a little bit