Adding logic for Admin & user