Now with OOP

2 clean classes implemented TODO: create a good gui