Adding tests for model associations