Updated Walter's code

and added some images