Updated the models slightly to reflect the type of data desired