Adding 'edit-mode' class to Edit if it's clicked