if else in dictionary comprehension