I also like working with music but can become unfocused. What seems to work for me is to listen to music in a foreign language because that way I can focus on my work and not the lyrics.