Improve trivial code style