Small code beautification