Remove some code. And add a class.
Add another.