test api, fixed model type
clean up imports
Merge pull request #6 from wmatyskiewicz/api

test api, fixed model type