Add doc comments