Prettified the code