add heart function - broken