The team behind the movie say Lee’s insights helped refine the heavily musical thriller