Remove unecesarry code.
Refactor.