if statement added