tried to get model to work but broke :(