just needing some touches in Models