Updated a statement in better code style