added model associations