Refactor using OOP

Here it is! #7, in code! :smiley: