Revert "updated code"

This reverts commit 32a69f45abec8cb31ea8bbd3f5c3b4e1b40559b3.