Fixes.
Clean up code.