Fixes and tidy up of plugin ui example