slight change to the final version