Prettify code 2