Treat depression after heart attack