More documentation and small UI changes