Small change to add def