If statement added