change if-else logic