Small change to functionality