Coding style

Made some improvements.