refactor code to use traits instead of the current strategy.

[calacuda]

I made this program early in my rust journey and it needs some polishing and refactoring. Including using traits.