Remove unused code.
Add test