Added more lines

Didn't work the first time