OOP version of basic program
Working OOP version of test.py