Got the model stuff basically working