added hearts feature