Fix style for code