adding if statment