List comprehension in edit employee