WIP: unit testing

unit test all models