Code style fixes in 2 lines