Fixing doc comments