added some more code relating to module 3 + start of module 4