Restyle code