Revert "Updated code"

This reverts commit e4be1216bbfee19b80e05ef593a522ba5562dcee.