Revert "updated code"

This reverts commit 3fcb796b2b969279d23157500642311c73c08889.