Added heart functionality