* Refactor the code, make things more OOP-style.
* Slight adjustment to the UI.