Improvements to the dead code lint

Awesome, thanks!