added another if statement and 2 more variables