Slight update to displayed code;