Added Sam's 2 lines of code