Add some go doc comments