Adding icon and removing div not needed