added if statement for some extra fun

added if statement for some extra fun