Clean up music code