Add correct code comment