Refactoring.
Fix shit.