Fix code stylings