adding If else statement