Added lines

Added three lines of code to the project
Merge pull request #1 from stuarth1/patch-1

Added lines