Add if condition