Added lines

Added three lines of code to the project