Adding some lines of code