Added a couple of lines