Improve the reference button