i addedd a couple of lines