Added more variables

Added more variables for the first if statement