Small changes

How dare you change my coding style. :P