Revert "Revert "Updated Code""

This reverts commit d0f235ee2b2fce1ebf5253864f853558e0e08eb5.