modidy the if/else statment