Add two lines of code