Add --keep-classes option

Closes #8 (closed).

I've added the option and tested the program locally. If you don't like this PR, I'll just use my fork personally. Any comments are welcome.