Little code style cleanup