Plot labels
Fix bug where state was not being properly closed over