got a little messy in Models