Add if/elsif/else statement