Updated code for import button