Various updates to the UI, fix missing files in build.