Revert "Updated code"

This reverts commit c784db5fd42a47f6a11c129231b05754e1ff7199.