Added new if statements

added new if statements to complete the script