change code to follow conventions
small change