Updated figure and fixed references