Add more model and prepare for unit test.
Using SqlAlchemy for modelling.