Beginning to add unit tests and figuring out peewee entities.