this is what i did for demo 4