Testing

I've added testing for the models.