fix song styling
simplify code