Add more doc comments