add if/else statements