Masked Gradient Descent for Localized Knowledge Editing - Coding problem | OneNoughtOne