Code style corrections