Various UI improvements, custom nodes working, but need more tests