Added code to get the models working