Prettified some code