small change to functionality