Small fixes

Documentation updates and code cleanup