Fix code prettify