Code beautified a little bit.