Models and such pretty much done, need to connect to view