Added If statements