Added a model class along with its unit tests.
Changed the user interface a bit.