The Wine and Spirit Trade Association has said it will work with the government on its proposal to implement a mandatory code of practice for retailers selling alcohol.