Code style correction