fixing stephen's code