Updated code.

Removed unneed code.