Doc fixes
Add some doc in code