Modify code for showing a heart