Small code beautifying