Code comments and documentation cleanup