Modification to Code