Add some (missing) comments