Add generate hearts to restart