Usually with headphones plugged in I try to stay away from music and switch to some apps like white noise machines or sites like asoftmurmur.com

Music can sometimes also lead to distraction itself unless it is classical or ambient. This switch to ambient noise generators has led to a significant change in productivity and focus personally and I do thoroughly advocate it.