Move lab 02 previous code to OOP