updated couple of lines