Add some doc and code comments