Put all logic in a class (OOP)