almost complete

complete model building
need to modify test.py