Add key mapping for class declarations