added heart feature