Improve style and add a reference