another if statement added to the bottom of the code.