Finally working on OOP